# -*- coding: utf-8 -*-
import pandas as pd
from database.database import etf_data_download
from strategy.model import feature_generator, LSTMTrainer

pd.set_option('display.width', 300)


if __name__ == "__main__":
    data = etf_data_download()
    df = data.load_data(code="513180", start_date="20240801", end_date="20240812")

    # 选取非日期和时间列
    model = LSTMTrainer(df)
    model.train()